import csv
from matplotlib import pyplot as plt
from datetime import datetime

# 使用自带的csv模块分析csv数据
filename = 'death_valley_2014.csv'
with open(filename) as f:
    reader = csv.reader(f)
    header_row = next(reader)
    # print(header_row)
    # 使用枚举的方式打印数据
    for index, column_header in enumerate(header_row):
        print(index, column_header)
    # 获取到最高气温，加上int转换变成数据，而不是string
    # 获取格式化的日期
    dates, highs, lows = [], [], []
    for row in reader:
        try:
            current_date = datetime.strptime(row[0], '%Y-%m-%d')
            high = int(row[1])
            low = int(row[3])
        except ValueError:
            print(current_date, 'missing-data')
        else:
            dates.append(current_date)
            highs.append(high)
            lows.append(low)

    # 绘图
    fig = plt.figure(dpi=128, figsize=(10, 6))
    plt.plot(dates, highs, c='red', alpha=0.5)
    plt.plot(dates, lows, c='blue', alpha=0.5)
    plt.fill_between(dates, highs, lows, facecolor='blue', alpha=0.1)
    # 设置图表标题
    plt.title("Daily high and low temperatures - 2014", fontsize=24)
    # 设置x、y轴的文字以及字体大小
    plt.xlabel("", fontsize=16)
    plt.ylabel("Temperature(F)", fontsize=16)
    # 设置刻度标记的大小
    plt.tick_params(axis='both', which='major', labelsize=16)
    plt.show()
